import torch
import torch.nn as nn
import json


class Config():
    def __init__(self):
        self.device = "cuda" if torch.cuda.is_available() else 'cpu'
        self.train_path = 'D:/HeiMa/pythonProject3/NER/data/data/train.txt'
        self.vocab_path = 'D:/HeiMa/pythonProject3/NER/vocab/vocab.txt'
        self.embedding_dim = 300
        self.epochs = 3
        self.batch_size = 8
        self.hidden_dim = 256
        self.lr = 2e-1
        self.dropout = 0.2
        self.model = "BiLSTM_CRF"  # 可以只用"BiLSTM"
        self.tag2id = json.load(open('D:/HeiMa/pythonProject3/NER/data/data/tag2id.json'))


if __name__ == '__main__':
    config = Config()
    print(config.tag2id)
